package com.ruoyi.hospital.utils;

import sun.misc.BASE64Decoder;

import javax.crypto.Cipher;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;
import javax.xml.bind.DatatypeConverter;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.*;


/**
 * AES加密128位CBC模式工具类
 */
public class AESUtils {
	
    //解密密钥(自行随机生成)
    public static final String KEY = "prod_doc_data_16";//密钥key
    public static final String IV  = "0102030405060708";//向量iv
	
    //认证密钥(自行随机生成)
    public static final String AK = "s2ip9g3y3bjr5zz7ws6kjgx3ysr82zzw";//AccessKey
    public static final String SK = "uv8zr0uen7aim8m7umcuooqzdv8cbvtf";//SecretKey
	
    //加密
    public static String Encrypt(String content) throws Exception {  
        byte[] raw = KEY.getBytes("utf-8");  
        SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");  
        Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");//"算法/模式/补码方式"
        //使用CBC模式，需要一个向量iv，可增加加密算法的强度    
        IvParameterSpec ips = new IvParameterSpec(IV.getBytes());
        cipher.init(Cipher.ENCRYPT_MODE, skeySpec, ips);  
        byte[] encrypted = cipher.doFinal(content.getBytes());

        //将字符串转换为Hex编码

        String hex = DatatypeConverter. printHexBinary(encrypted);

        //return new BASE64Encoder().encode(encrypted);
        return hex;
    }  
	
    //解密  
    public static String Decrypt(String content) throws Exception {  
        try {   
            byte[] raw = KEY.getBytes("utf-8");  
            SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");  
            Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");  
            IvParameterSpec ips = new IvParameterSpec(IV.getBytes());  
            cipher.init(Cipher.DECRYPT_MODE, skeySpec, ips);


            // 将Hex编码转换为字符串
            byte[] bytes = DatatypeConverter.parseHexBinary(content);
            String str = new String(bytes);

            byte[] encrypted1 = new BASE64Decoder().decodeBuffer(content);
            encrypted1 = bytes;
            try {  
                byte[] original = cipher.doFinal(encrypted1);  
                String originalString = new String(original);  
                return originalString;  
            } catch (Exception e) {  
                System.out.println(e.toString());  
                return null;  
            }  
        } catch (Exception ex) {  
            System.out.println(ex.toString());  
            return null;  
        }  
    } 
	
    //获取认证签名(身份认证需要)
    public static String getSign(String currentTime) throws Exception { 
	String sign = "";
	Map<String, Object> map=new HashMap<String,Object>();
        map.put("ak",AK);
        map.put("sk",SK);
        map.put("ts",currentTime);
	//获取 参数字典排序后字符串  
        String decrypt = getOrderMap(map);
	try {  
            //指定sha1算法  
            MessageDigest digest = MessageDigest.getInstance("SHA-1");  
            digest.update(decrypt.getBytes());  
            //获取字节数组  
            byte messageDigest[] = digest.digest();  
            // Create Hex String  
            StringBuffer hexString = new StringBuffer();  
            // 字节数组转换为十六进制数  
            for (int i = 0; i < messageDigest.length; i++) {  
                String shaHex = Integer.toHexString(messageDigest[i] & 0xFF);  
                if (shaHex.length() < 2) {  
                    hexString.append(0);  
                }  
                hexString.append(shaHex);  
            }  
            sign = hexString.toString();  
        } catch (NoSuchAlgorithmException e) {  
            e.printStackTrace();  
        } 
		return sign;
    } 
	
    //获取参数的字典排序  
    private static String getOrderMap(Map<String,Object> maps){  
    	List<String> paramNames = new ArrayList<String>();  
        for(Map.Entry<String,Object> entry : maps.entrySet()){  
            paramNames.add(entry.getValue().toString());  
        }  
        Collections.sort(paramNames); 
        StringBuilder paramStr = new StringBuilder();  
        for(String paramName : paramNames){  
            paramStr.append(paramName);    
        }  
        return paramStr.toString();  
    }  
 
}